tensor([[0.7100, 0.7100, 0.7100, 0.7100, 0.7100],
        [0.2900, 0.2900, 0.2900, 0.2900, 0.2900]], grad_fn=<RoundBackward1>)